package dhy.com.client;

import dhy.com.config.SequenceIdGenerator;
import dhy.com.config.protocol.MessageCodecSharable;
import dhy.com.config.protocol.ProcotolFrameDecoder;
import dhy.com.container.RpcContainerImpl;
import dhy.com.handler.RpcResponseMessageHandler;
import dhy.com.message.RpcRequestMessage;
import io.netty.bootstrap.Bootstrap;
import io.netty.channel.Channel;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioSocketChannel;
import io.netty.handler.logging.LogLevel;
import io.netty.handler.logging.LoggingHandler;
import io.netty.util.concurrent.DefaultPromise;
import lombok.extern.slf4j.Slf4j;

import java.lang.reflect.Proxy;
import java.util.List;

/**
 * <p>
 * 客户端通信管理器
 * </p>
 *
 * @author zdh
 */
@Slf4j
public abstract class AbstractRpcClientManager {

    /**
     * 是否开启rpc远程调用服务
     */
    protected boolean isEnableRpcServer=false;

    /**
     * <P>
     *     创建代理类,主要使用反向代理来创建代理类,并且发送给对应服务器掉用方法信息,获得返回值并保存在promise对象中
     * </P>
     */
   final public  <T> T getProxyService(Class<T> serviceClass) {
        //获取类加载器
        ClassLoader loader = serviceClass.getClassLoader();
        //获取要代理类的接口
        Class<?>[] interfaces = new Class[]{serviceClass};
        //生成代理对象---使用jdk的反向代理技术---所有代理对象方法的执行都会走第三个回调函数,来完成方法的执行
        Object o = Proxy.newProxyInstance(loader, interfaces, (proxy, method, args) -> {
            //将方法调用转换为 消息对象
            //每个消息都有唯一的消息标识
            int sequenceId = SequenceIdGenerator.nextId();
            RpcRequestMessage msg = new RpcRequestMessage(
                    //消息唯一标识
                    sequenceId,
                    //要执行类的类名
                    serviceClass.getSimpleName(),
                    //方法名
                    method.getName(),
                    //方法返回值类型
                    method.getReturnType(),
                    //方法的参数类型
                    method.getParameterTypes(),
                    //具体参数
                    args
            );
            //将消息对象发送出去
            getChannel().writeAndFlush(msg);

            //准备一个空 Promise 对象，来接收结果             指定 promise 对象异步接收结果线程
            DefaultPromise<Object> promise = new DefaultPromise<>(getChannel().eventLoop());
            //将promise放入集合中保存,待响应消息回来后再取出
            RpcResponseMessageHandler.PROMISES.put(sequenceId, promise);

            //异步等待结果
//            promise.addListener(future -> {
//                // 线程
//            });

            //同步待 promise 结果
            //await等待不会抛出异常,需要我们自己去判断
            promise.await();
            log.debug("当前结果计算返回");
            if (promise.isSuccess()) {
                // 调用正常
                return promise.getNow();
            } else {
                // 调用失败
                throw new RuntimeException(promise.cause());
            }
        });
        //返回代理对象
        return (T) o;
    }

    /**
     * 通道
     */
    private static Channel channel = null;
    /**
     * 同步锁
     */
    private static final Object LOCK = new Object();


    /**
     * <P>
     *     获取唯一的 channel 对象--采用单例模式中的双重校验完成
     * </P>
     */
   final protected   Channel getChannel() {
        if (channel != null) {
            return channel;
        }
        synchronized (LOCK) {
            if (channel != null) {
                return channel;
            }
            //初始化通道
            initChannel();
            return channel;
        }
    }


    /**
     * 初始化 channel 方法
     */
    private  void initChannel() {
        NioEventLoopGroup group = new NioEventLoopGroup();
        LoggingHandler LOGGING_HANDLER = new LoggingHandler(LogLevel.DEBUG);
        MessageCodecSharable MESSAGE_CODEC = new MessageCodecSharable();
        RpcResponseMessageHandler RPC_HANDLER = new RpcResponseMessageHandler();
        Bootstrap bootstrap = new Bootstrap();
        bootstrap.channel(NioSocketChannel.class);
        bootstrap.group(group);
        bootstrap.handler(new ChannelInitializer<SocketChannel>() {
            @Override
            protected void initChannel(SocketChannel ch) throws Exception {
                ch.pipeline().addLast(new ProcotolFrameDecoder());
              //  ch.pipeline().addLast(LOGGING_HANDLER);
                ch.pipeline().addLast(MESSAGE_CODEC);
                ch.pipeline().addLast(RPC_HANDLER);
            }
        });
        try {
            //建立连接前
            beforeBuildConnection();
           //默认不开启远程调用服务
            if(isEnableRpcServer)
           {
               channel = bootstrap.connect("localhost", 8080).sync().channel();
               //建立连接后
               afterBuildConnection();
               channel.closeFuture().addListener(future -> {
                   group.shutdownGracefully();
                   //关闭连接后
                   afterCloseConnection();
               });
           }
        } catch (Exception e) {
            log.error("client error", e);
        }
    }

    public  void afterCloseConnection() {
    }

    public  void afterBuildConnection() {
    }

    public  void beforeBuildConnection() {
    }

    public void startRPC()
    {
        getChannel();
    }



}
